Made memoize macro support multiple statements