Shubham Jain's submission of ATM | CodeChef