Checkpoint upgrade script