Updating to only use websockets