updated SHO.py