Add a callback for init