problem 2 python