add changes in question1.py