update readme and code