Add additional tests for memoize