interview question in python