assignment 1.py