Add python interview question 2