Update readme and add comments