Updated readme and commented some of the code