Improve memoization comment