Punit Solanki's submission of ATM | CodeChef