added checkpoint and gb for main