python interview question 3