updated checkpoint stuff
checkpoint implementation update