Shubham Srivastava's submission of ATM | CodeChef