Add checkpoint upgrade to migration PS script