Do problem 4 in Python.