Add option to save plugin to set a custom callback

Close #17 & #19