Switch to gorilla/websocket