Updated to use the new official Azure SDK