diff --git a/azure/README-zh.md b/azure/README-zh.md
index a1cf55f..03b83d0 100644
--- a/azure/README-zh.md
+++ b/azure/README-zh.md
@@ -8,9 +8,9 @@
 
  - Username for VPN and SSH （用户名）
  - Password for VPN and SSH （密码）
- - IPsec Pre-Shared Key （IPsec 预共享密钥）
+ - IPsec Pre-Shared Key for VPN （IPsec 预共享密钥）
  - Operating System Image （操作系统镜像，Debian 8 或 Ubuntu 16.04 LTS）
- - Virtual Machine Size （虚拟机大小，默认值： Standard_A0）
+ - Virtual Machine Size （虚拟机大小，默认值： Basic_A0）
 
 请单击以下按钮开始：
 
@@ -20,6 +20,7 @@
 
 ## 作者
 
+版权所有 (C) 2017 Lin Song   
 版权所有 (C) 2016 [Daniel Falkner](https://github.com/derdanu)
 
 ## 屏幕截图
diff --git a/azure/README.md b/azure/README.md
index 697f219..ed63b5a 100644
--- a/azure/README.md
+++ b/azure/README.md
@@ -8,9 +8,9 @@ Customizable with the following options:
 
  - Username for VPN and SSH
  - Password for VPN and SSH
- - IPsec Pre-Shared Key
+ - IPsec Pre-Shared Key for VPN
  - Operating System Image (Debian 8 or Ubuntu 16.04 LTS)
- - Virtual Machine Size (Default: Standard_A0)
+ - Virtual Machine Size (Default: Basic_A0)
 
 Press this button to start:
 
@@ -18,8 +18,9 @@ Press this button to start:
     <img src="../docs/images/azure-deploy-button.png" alt="Deploy to Azure" />
 </a>
 
-## Author
+## Authors
 
+Copyright (C) 2017 Lin Song   
 Copyright (C) 2016 [Daniel Falkner](https://github.com/derdanu)
 
 ## Screenshot
diff --git a/azure/azuredeploy.json b/azure/azuredeploy.json
index 9e08549..b330df6 100644
--- a/azure/azuredeploy.json
+++ b/azure/azuredeploy.json
@@ -6,36 +6,41 @@
       "type": "string",
       "minLength": 1,
       "metadata": {
-        "description": "User name for SSH and VPN"
+        "description": "Username for VPN and SSH"
       }
     },
     "password": {
       "type": "securestring",
       "metadata": {
-        "description": "User password for SSH and VPN"
+        "description": "Password for VPN and SSH"
       }
     },
    "preSharedKey": {
       "type": "securestring",
       "metadata": {
-        "description": "Pre-Shared Key for VPN"
+        "description": "IPsec Pre-Shared Key for VPN"
       }
     },
     "image": {
       "type": "string",
       "allowedValues": [
-        "ubuntu",
-        "debian"
+        "ubuntu1604",
+        "debian8"
       ],
-      "defaultValue": "debian",
+      "defaultValue": "debian8",
       "metadata": {
-        "description": "OS to use. Debian or Ubuntu"
+        "description": "OS to use. Debian 8 or Ubuntu 16.04 LTS"
       }
     },
     "VMSize": {
       "type": "string",
-      "defaultValue": "Standard_A0",
+      "defaultValue": "Basic_A0",
       "allowedValues": [
+        "Basic_A0",
+        "Basic_A1",
+        "Basic_A2",
+        "Basic_A3",
+        "Basic_A4",
         "Standard_A0",
         "Standard_A1",
         "Standard_A2",
@@ -44,11 +49,6 @@
         "Standard_A5",
         "Standard_A6",
         "Standard_A7",
-        "Basic_A0",
-        "Basic_A1",
-        "Basic_A2",
-        "Basic_A3",
-        "Basic_A4",
         "Standard_D1",
         "Standard_D2",
         "Standard_D3",
@@ -71,13 +71,13 @@
     "vhdStorageType": "Standard_LRS",
     "vnetId": "[resourceId('Microsoft.Network/virtualNetworks', variables('virtualNetworkName'))]",
     "SubnetRef": "[concat(variables('vnetId'), '/subnets/', variables('subnetName'))]",
-    "ubuntu": {
+    "ubuntu1604": {
       "publisher": "Canonical",
       "offer": "UbuntuServer",
       "sku": "16.04-LTS",
       "version": "latest"
     },
-    "debian": {
+    "debian8": {
       "publisher": "credativ",
       "offer": "Debian",
       "sku": "8",
@@ -210,9 +210,10 @@
         "[concat('Microsoft.Compute/virtualMachines/', variables('vmName'))]"
       ],
       "properties": {
-        "publisher": "Microsoft.OSTCExtensions",
-        "type": "CustomScriptForLinux",
-        "typeHandlerVersion": "1.3",
+        "publisher": "Microsoft.Azure.Extensions",
+        "type": "CustomScript",
+        "typeHandlerVersion": "2.0",
+        "autoUpgradeMinorVersion": true,
         "settings": {
           "fileUris": [ "[variables('installScriptURL')]" ],
           "commandToExecute": "[variables('installCommand')]"
diff --git a/azure/custom_deployment_screenshot.png b/azure/custom_deployment_screenshot.png
index beb5911..70d62f2 100644
Binary files a/azure/custom_deployment_screenshot.png and b/azure/custom_deployment_screenshot.png differ
diff --git a/azure/install.sh b/azure/install.sh
index e61c903..4174f5f 100644
--- a/azure/install.sh
+++ b/azure/install.sh
@@ -1,13 +1,7 @@
-#!/bin/bash
+#!/bin/sh
+
 export VPN_IPSEC_PSK=$1
 export VPN_USER=$2
 export VPN_PASSWORD=$3
 
-# Debian on Azure has no lsb_release installed.
-if ! [[ -x "/usr/bin/lsb_release" ]]
-then
-    apt-get update
-    apt-get install -y lsb-release
-fi
-
-wget https://git.io/vpnsetup -O vpnsetup.sh && sh vpnsetup.sh
\ No newline at end of file
+wget https://git.io/vpnsetup -O vpnsetup.sh && sh vpnsetup.sh