Updated Homework_2.py