Adds way to register custom callbacks