fix readme and add comments