Added websockets integration