Some minor updates in comment and code.