updated milestone2.py