Vivek Gupta's submission of ATM | CodeChef