Arjun P's submission of ATM | CodeChef