Python List Exercise Geeksforgeeks