python problem 1

First assignment