Vaibhav Puri's submission of ATM | CodeChef