Update readme
Polish documentation and comments