Tweaks to PSF plot