Added checkpoint to run model