Added websocket server code