update homework from py
update homework from py