code update for readme