adding the change for ex1.py file