Vivek Kumar's submission of ATM | CodeChef