Add a callback for init

Just my hack, if you think you have a better way.