Added memoization and updated screenshot