diff --git a/README.md b/README.md
index c8defb1..dd59df5 100644
--- a/README.md
+++ b/README.md
@@ -3,7 +3,7 @@ Deploys a group of Virtual Machines exposed to a public IP via a Load Balancer
 
 This Terraform module deploys a Virtual Machines Scale Set in Azure and opens the specified ports on the loadbalancer for external access and returns the id of the VM scale set deployed.
 
-This module requires a network and loadbalancer to be provider separately. You can provision them with the "Azure/network/azurerm" and "Azure/loadbanacer/azurerm" modules.
+This module requires a network and loadbalancer to be provider separately. You can provision them with the "Azure/network/azurerm" and "Azure/loadbalancer/azurerm" modules.
 
 [![Build Status](https://travis-ci.org/Azure/terraform-azurerm-computegroup.svg?branch=master)](https://travis-ci.org/Azure/terraform-azurerm-computegroup)
 
@@ -14,7 +14,7 @@ Using the `vm_os_simple`:
 
 ```hcl 
 provider "azurerm" {
-  version = "~> 0.3"
+  version = "~> 1.0"
 }
 
 variable "resource_group_name" {
@@ -22,19 +22,21 @@ variable "resource_group_name" {
 }
 
 module "network" {
-    source = "Azure/network/azurerm"
-    location = "westus"
+    source              = "Azure/network/azurerm"
+    location            = "westus"
     resource_group_name = "${var.resource_group_name}"
   }
 
 module "loadbalancer" {
-  source = "Azure/loadbalancer/azurerm"
+  source              = "Azure/loadbalancer/azurerm"
   resource_group_name = "${var.resource_group_name}"
-  location = "westus"
-  prefix = "terraform-test"
-  "lb_port" {
-      http = [ "80", "Tcp", "80"]
-  }
+  location            = "westus"
+  prefix              = "terraform-test"
+  lb_port             = { 
+                          http  = ["80", "Tcp", "80"]
+                          https = ["443", "Tcp", "443"]
+                          #ssh   = ["22", "Tcp", "22"]
+                        }
 }
 
 module "computegroup" { 
@@ -50,10 +52,6 @@ module "computegroup" {
     vnet_subnet_id      = "${module.network.vnet_subnets[0]}"
     load_balancer_backend_address_pool_ids = "${module.loadbalancer.azurerm_lb_backend_address_pool_id}"
     cmd_extension       = "sudo apt-get -y install nginx"
-    lb_port             = { 
-                            http = ["80", "Tcp", "80"]
-                            https = ["443", "Tcp", "443"]
-                          }
     tags                = {
                             environment = "dev"
                             costcenter  = "it"
@@ -71,7 +69,7 @@ Using the `vm_os_publisher`, `vm_os_offer` and `vm_os_sku`
 ```hcl
 
 provider "azurerm" {
-  version = "~> 0.3"
+  version = "~> 1.0"
 }
 
 variable "resource_group_name" {
@@ -79,19 +77,21 @@ variable "resource_group_name" {
 }
 
 module "network" {
-    source = "Azure/network/azurerm"
-    location = "westus"
+    source              = "Azure/network/azurerm"
+    location            = "westus"
     resource_group_name = "${var.resource_group_name}"
   }
 
 module "loadbalancer" {
-  source = "Azure/loadbalancer/azurerm"
+  source              = "Azure/loadbalancer/azurerm"
   resource_group_name = "${var.resource_group_name}"
-  location = "westus"
-  prefix = "terraform-test"
-  "lb_port" {
-      http = [ "80", "Tcp", "80"]
-  }
+  location            = "westus"
+  prefix              = "terraform-test"
+  lb_port             = { 
+                          http  = ["80", "Tcp", "80"]
+                          https = ["443", "Tcp", "443"]
+                          #ssh   = ["22", "Tcp", "22"]
+                        }
 }
 
 module "computegroup" {
@@ -109,10 +109,6 @@ module "computegroup" {
     vnet_subnet_id      = "${module.network.vnet_subnets[0]}"
     load_balancer_backend_address_pool_ids = "${module.loadbalancer.azurerm_lb_backend_address_pool_id}"
     cmd_extension       = "sudo apt-get -y install nginx"
-    lb_port             = { 
-                            http = ["80", "Tcp", "80"]
-                            https = ["443", "Tcp", "443"]
-                          }
     tags                = {
                             environment = "dev"
                             costcenter  = "it"
@@ -129,7 +125,7 @@ The module does not expose direct access to each node of the VM scale set for se
 
 ```hcl
 provider "azurerm" {
-  version = "~> 0.3"
+  version = "~> 1.0"
 }
 
 variable "resource_group_name" {
@@ -149,11 +145,12 @@ module "network" {
 module "loadbalancer" {
   source = "Azure/loadbalancer/azurerm"
   resource_group_name = "${var.resource_group_name}"
-  location = "${var.location}"
-  prefix = "terraform-test"
-  "lb_port" {
-      http = [ "80", "Tcp", "80"]
-  }
+  location            = "${var.location}"
+  prefix              = "terraform-test"
+  lb_port             = { 
+                          http  = ["80", "Tcp", "80"]
+                          https = ["443", "Tcp", "443"]
+                        }
 }
 
 module "computegroup" {
@@ -171,9 +168,6 @@ module "computegroup" {
     vnet_subnet_id      = "${module.network.vnet_subnets[0]}"
     load_balancer_backend_address_pool_ids = "${module.loadbalancer.azurerm_lb_backend_address_pool_id}"
     cmd_extension       = "sudo apt-get -y install nginx"
-    lb_port             = {
-                            http = ["80", "Tcp", "80"]
-                          }
     tags                = {
                             environment = "codelab"
                           }
diff --git a/main.tf b/main.tf
index 05b4de7..4bcb317 100644
--- a/main.tf
+++ b/main.tf
@@ -1,5 +1,5 @@
 provider "azurerm" {
-  version = "~> 0.3"
+  version = "~> 1.0"
 }
 
 module "os" {
@@ -77,13 +77,13 @@ resource "azurerm_virtual_machine_scale_set" "vm-linux" {
 
   extension {
     name                 = "vmssextension"
-    publisher            = "Microsoft.OSTCExtensions"
-    type                 = "CustomScriptForLinux"
-    type_handler_version = "1.2"
+    publisher            = "Microsoft.Azure.Extensions"
+    type                 = "CustomScript"
+    type_handler_version = "2.0"
 
     settings = <<SETTINGS
     {
-        "commandToExecute": "${var.cmd_extension}"
+        "script": "${base64encode("${var.cmd_extension}")}"
     }
     SETTINGS
   }
@@ -146,7 +146,7 @@ resource "azurerm_virtual_machine_scale_set" "vm-windows" {
     name                 = "vmssextension"
     publisher            = "Microsoft.Compute"
     type                 = "CustomScriptExtension"
-    type_handler_version = "1.8"
+    type_handler_version = "1.9"
 
     settings = <<SETTINGS
     {
diff --git a/test/integration/fixtures/main.tf b/test/integration/fixtures/main.tf
index 0a4a8b8..9af5111 100644
--- a/test/integration/fixtures/main.tf
+++ b/test/integration/fixtures/main.tf
@@ -1,5 +1,5 @@
 provider "azurerm" {
-  version = "~> 0.3"
+  version = "~> 1.0"
 }
 
 module "network" {
diff --git a/variables.tf b/variables.tf
index bdaac88..0c48764 100644
--- a/variables.tf
+++ b/variables.tf
@@ -91,14 +91,9 @@ variable "load_balancer_backend_address_pool_ids" {
   description = "The id of the backend address pools of the loadbalancer to which the VM scale set is attached"
 }
 
-variable "lb_port" {
-  description = "Protocols to be used for lb health probes and rules. [frontend_port, protocol, backend_port]"
-  default     = {}
-}
-
 variable "cmd_extension" {
   description = "Command to be excuted by the custom script extension"
-  default     = ""
+  default     = "echo hello"
 }
 
 variable "tags" {