problem 2, python