Improvements

Add minor performance improvement, and clarify the use of "identity" lambda via comment