modified:   Assignment2.py