Rishi Chawla's submission of ATM | CodeChef