assignment1.py