Modify readme and fix bug