small code improvement in README demo