Create assignment5_v1.py