Adding interface to change the callback post-initialization

Adding interface to change the callback post-initialization #9