python interview preparation question