Modify Readme and fix comments of code