Add web socket implementation