- added comments to simulate_psi.py