my home work for python