Update readme
Fix bugs