modified the psh.py file