Added tests for checking the callback, and allowing the ability to unset the callback.