Yi Feng Huang's submission of ATM | CodeChef