minor changes in comment and diagram