Adding websockets support