Rohan Garg's submission of ATM | CodeChef