change to Checkpoint class

according to Professor Herman's pointers, change checkpoint class