clean up comments
tidying up
modifying print statement
better checkpointing
update