Ankitanshu Swaroop's submission of ATM | CodeChef