Add connection status callback