Add initial websocket support