End Code Function Python Questions Geeksforgeeks