Add a callback option

[appybara13]

Allowing the user to set a callback, that gets called on when an allocation/free happens, would help with debugging.