Create homework4.py