Update readme, and fix minor bug