Update homework.py

http://www.codeskulptor.org/#user34_NsI6bIrSa6_6.py