Tidy up changes to pbs.py file.