Shivam Khandelwal 's submission of ATM | CodeChef