update readme and fix bug