this is my python homework