Add authentication to websocket connections