Arun Kumar's submission of ATM | CodeChef