Amit Dwivedi's submission of ATM | CodeChef