Minor updates to psf_display