Umang Goel's submission of ATM | CodeChef