modify code example of readme