Update First_assignment.py