Add websocket implementation