Create HW5.py

In progress