Add websocket connection check