Add support for getting all checkpoint keys