Move compute_psl.py into python package