Update checkpoint test

* Fix some build warnings
* Add some more comments