Aniruddh Kanojia's submission of ATM | CodeChef