Create assignment_5-1.py