jay patel's submission of Train or Walk | CodeChef