Working on setting up data
Added checkpoints