problem 2 python in progress