Refactored
Updated memoize import.