Create final_assignment.py