Andrew's first python program