Add onDisconnect callback