Geeksforgeeks Python Problem Solving