Codechef IITK1P02 submission