updated py

updated python file