Updated job to support checkpoint