python homwork