diff --git a/README.md b/README.md
index 362bfc3..a3de170 100644
--- a/README.md
+++ b/README.md
@@ -26,11 +26,15 @@ Alternatively, you can include `$WomensRefugeShieldButton` anywhere in your temp
 
 ![Button in the footer](img/screenshots/button_in_footer.PNG?raw=true)
 
+## Versioning
+
+- version 1.x is for SilverStripe 3
+- version 2.x is for SilverStripe 4
+
 # Options
 
 You can view all the options on the [Women's Refuge Shielded Site website](https://shielded.co.nz/).
 
-
 # TODO
 
  - responsiveness
diff --git a/_config.php b/_config.php
index cfb4525..b3d9bbc 100644
--- a/_config.php
+++ b/_config.php
@@ -1,3 +1 @@
 <?php
-
-define('SS_WOMENS_REFUGE_DIR', basename(dirname(__FILE__)));
diff --git a/_config/extensions.yml b/_config/extensions.yml
index c0a9fe7..8955e1d 100644
--- a/_config/extensions.yml
+++ b/_config/extensions.yml
@@ -1,6 +1,6 @@
-SiteConfig:
+SilverStripe\SiteConfig\SiteConfig:
   extensions:
-    - WomensRefugeShieldSiteConfigExtension
-ContentController:
+    - AndrewAndante\WomensRefugeShield\SiteConfigExtension
+SilverStripe\CMS\Controllers\ContentController:
   extensions:
-    - WomensRefugeShieldControllerExtension
+    - AndrewAndante\WomensRefugeShield\ControllerExtension
diff --git a/code/WomensRefugeShieldControllerExtension.php b/code/ControllerExtension.php
similarity index 54%
rename from code/WomensRefugeShieldControllerExtension.php
rename to code/ControllerExtension.php
index 62fd5e9..91c44b8 100644
--- a/code/WomensRefugeShieldControllerExtension.php
+++ b/code/ControllerExtension.php
@@ -1,26 +1,32 @@
 <?php
 
+namespace AndrewAndante\WomensRefugeShield;
+
+use SilverStripe\CMS\Controllers\ContentController;
+use SilverStripe\Core\Extension;
+use SilverStripe\SiteConfig\SiteConfig;
+use SilverStripe\View\Requirements;
+
 
 /**
- * Class WomensRefugeShieldControllerExtension
+ * Class AndrewAndante\WomensRefugeShield\ControllerExtension
  *
  * @property ContentController $owner
  */
-class WomensRefugeShieldControllerExtension extends Extension
+class ControllerExtension extends Extension
 {
 
     public function onAfterInit()
     {
         $config = SiteConfig::current_site_config();
-        $jsDir = SS_WOMENS_REFUGE_DIR . DIRECTORY_SEPARATOR . 'javascript' . DIRECTORY_SEPARATOR;
         switch ($config->ShieldCode) {
             case 1:
                 Requirements::javascript('https://d3f5l8ze0o4j2m.cloudfront.net/m87/k33spt.js');
-                Requirements::javascript($jsDir.'large_tab.js');
+                Requirements::javascript('andrewandante/womens-refuge-shield: javascript/large_tab.js');
                 break;
             case 2:
                 Requirements::javascript('https://d3f5l8ze0o4j2m.cloudfront.net/m87/k33spt.js');
-                Requirements::javascript($jsDir.'small_tab.js');
+                Requirements::javascript('andrewandante/womens-refuge-shield: javascript/small_tab.js');
                 break;
         }
     }
diff --git a/code/WomensRefugeShieldSiteConfigExtension.php b/code/SiteConfigExtension.php
similarity index 63%
rename from code/WomensRefugeShieldSiteConfigExtension.php
rename to code/SiteConfigExtension.php
index 98c3b3f..73e905a 100644
--- a/code/WomensRefugeShieldSiteConfigExtension.php
+++ b/code/SiteConfigExtension.php
@@ -1,12 +1,19 @@
 <?php
 
+namespace AndrewAndante\WomensRefugeShield;
+
+use SilverStripe\Forms\DropdownField;
+use SilverStripe\Forms\FieldList;
+use SilverStripe\ORM\DataExtension;
+use SilverStripe\SiteConfig\SiteConfig;
+
 /**
- * Class WomensRefugeShieldSiteConfigExtension
+ * Class AndrewAndante\WomensRefugeShield\SiteConfigExtension
  *
  * @property SiteConfig $owner
  * @property int $Pridecode
  */
-class WomensRefugeShieldSiteConfigExtension extends DataExtension
+class SiteConfigExtension extends DataExtension
 {
 
     private static $db = array(
diff --git a/composer.json b/composer.json
index 4ae537e..864153b 100644
--- a/composer.json
+++ b/composer.json
@@ -1,7 +1,7 @@
 {
   "name": "andrewandante/womens-refuge-shield",
   "description": "Add the Women's refuge shield to your site",
-  "type": "silverstripe-module",
+  "type": "silverstripe-vendormodule",
   "keywords": ["silverstripe", "womens", "women", "refuge", "shield"],
   "license": "BSD-3-Clause",
   "authors": [
@@ -13,8 +13,20 @@
   ],
   "require":
   {
-    "silverstripe/framework": "~3.0",
-    "silverstripe/siteconfig": "~3.0",
-    "silverstripe/cms": "~3.0"
+    "silverstripe/framework": "^4.0",
+    "silverstripe/siteconfig": "^4.0",
+    "silverstripe/cms": "^4.0"
+  },
+  "extra": {
+    "expose": [
+        "css",
+        "javascript",
+        "img/logos"
+    ]
+  },
+  "autoload": {
+    "psr-4": {
+      "AndrewAndante\\WomensRefugeShield\\": "code/"
+    }
   }
 }
diff --git a/img/logos/custom-logo.png b/img/logos/custom-logo.png
new file mode 100644
index 0000000..c8ed1e4
Binary files /dev/null and b/img/logos/custom-logo.png differ
diff --git a/templates/WomensRefugeShieldButton.ss b/templates/WomensRefugeShieldButton.ss
index f30b066..e257a88 100644
--- a/templates/WomensRefugeShieldButton.ss
+++ b/templates/WomensRefugeShieldButton.ss
@@ -1,7 +1,7 @@
 <% require javascript("https://d3f5l8ze0o4j2m.cloudfront.net/m87/k33spt.js") %>
-<% require javascript("womens-refuge-shield/javascript/button.js") %>
-<% require css("womens-refuge-shield/css/button.css") %>
+<% require javascript("andrewandante/womens-refuge-shield: javascript/button.js") %>
+<% require css("andrewandante/womens-refuge-shield: css/button.css") %>
 
 <a href="#" id="womens-refuge-shield-button">
-    <img src="womens-refuge-shield/img/custom-logo.png">
+    <img src="$ResourceURL(andrewandante/womens-refuge-shield: img/logos/custom-logo.png)">
 </a>