Make web socket optional