Add changes in code 1.py