Added callback for controller state changes