assignment-2b.py