working on psf checkpoint