Create assignment5_v2.py