modified readme and brought code closer to checkpoint.