Added additional_key to memoization function

Fixes #29