Refactor memoized_method registratio to use default proc