Add websocket reconnection