Update readme
Fix bug