Update Assignment1.py


+plot color