my first python program from the book