adding websocket
bug fix, websocket work