put all code in homework.py answered question2