homework.py and test_homework.py are added based on the method structure provided by X-Treem