Sujit Jha's submission of ATM | CodeChef