answer 1 in python