Create 4. assignment.py