Shivam Bajaj's submission of ATM | CodeChef