Taranpreet Singh's submission of ATM | CodeChef