Yash Kesharwani's submission of ATM | CodeChef