Nisarg Shah's submission of ATM | CodeChef