include new tests for memoization