fix bug, updated minor change in README example