Add code to read checkpoint.