Add websocket handler