Update readme and fix