Update readme, fix bugs