CodeChef Assignment