edits to NX01_psr.py