Added callback register method