Update homework.py

Margus, don't steal :)