I added function in pr9.py