Added on_communication_success callback