Homework for python