diff --git a/LICENSE b/LICENSE
index 4966d1f..ee6ef85 100644
--- a/LICENSE
+++ b/LICENSE
@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2021 Bjarke Berg
+Copyright (c) 2021 Nik Rimington
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
diff --git a/README.md b/README.md
index a296d35..61e2a35 100644
--- a/README.md
+++ b/README.md
@@ -19,6 +19,24 @@ To [install from NuGet](https://www.nuget.org/packages/HotChilli.Umbraco.PageNot
 
     PM> Install-Package HotChilli.Umbraco.PageNotFound
 
+## How to use
+
+*Improved documentation will come in time*
+
+Once the package has been installed there are two approaches to configuration. Out of the box, it will be loaded via an Umbraco Composer, but if you wish to have more control over when it loads, there is an extension method for `IUmbracoBuilder` called `UsePageNotFoundManager`.
+
+Once installed and running, Page Not Found Manager extends the "Do Something Else" context menu with a new entry for picking a 404 page off of a content node.
+
+This allows for multiple different 404 pages to be configured for different parts of the website.
+
+## Future Plans
+
+There is currently a plan to investigate and see if this same 404 page can be served for missing media items. In IIS, this would fall back to the default 404 page from IIS but I'm yet to understand this pipeline and if this is possible - I'm open to PR's and suggestions on improvements as well.
+
+## Special Thanks
+
+This package is a port of [PageNotFoundManager](https://github.com/TimGeyssens/UmbracoPageNotFoundManager) by [Tim Geyssens](https://twitter.com/timgeyssens) and was created with his permission. The code based used as a point of reference was taken prior to the Licesencing Model change.
+
 ## Credits
 
 The logo uses [404](https://thenounproject.com/term/404/3283006 ) from the [Noun Project](https://thenounproject.com) by [Ilham Fitrotul Hayat](https://thenounproject.com/fhilham), licensed under [CC BY 3.0 US](https://creativecommons.org/licenses/by/3.0/us/).
diff --git a/src/HC.PageNotFoundManager.Core/Backoffice/MenuRenderingNotificationHandler.cs b/src/HC.PageNotFoundManager.Core/Backoffice/MenuRenderingNotificationHandler.cs
index 735eabb..5ed0b6c 100644
--- a/src/HC.PageNotFoundManager.Core/Backoffice/MenuRenderingNotificationHandler.cs
+++ b/src/HC.PageNotFoundManager.Core/Backoffice/MenuRenderingNotificationHandler.cs
@@ -37,7 +37,7 @@ public void Handle(MenuRenderingNotification notification)
 
             var menuItem = new MenuItem(Constants.MenuAlias, Constants.MenuLabel)
             {
-                Icon = Umbraco.Cms.Core.Constants.Icons.Content,
+                Icon = "alert red",
                 SeparatorBefore = true
             };
 
diff --git a/src/HC.PageNotFoundManager.Core/HC.PageNotFoundManager.Core.csproj b/src/HC.PageNotFoundManager.Core/HC.PageNotFoundManager.Core.csproj
index cf497ed..345ef9c 100644
--- a/src/HC.PageNotFoundManager.Core/HC.PageNotFoundManager.Core.csproj
+++ b/src/HC.PageNotFoundManager.Core/HC.PageNotFoundManager.Core.csproj
@@ -3,6 +3,7 @@
     <PropertyGroup>
         <AssemblyVersion>1.0.0</AssemblyVersion>
         <Version>1.0.0</Version>
+        <VersionSuffix>beta-004</VersionSuffix>
         <InformationalVersion>1.0.0</InformationalVersion>
         <TargetFramework>net5.0</TargetFramework>
         <Authors>Nik Rimington</Authors>
@@ -22,11 +23,11 @@
     </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Umbraco.Cms.Core" Version="9.0.0-rc001" />
-    <PackageReference Include="Umbraco.Cms.Infrastructure" Version="9.0.0-rc001" />
-    <PackageReference Include="Umbraco.Cms.Web.BackOffice" Version="9.0.0-rc001" />
-    <PackageReference Include="Umbraco.Cms.Web.Common" Version="9.0.0-rc001" />
-    <PackageReference Include="Umbraco.Cms.Web.Website" Version="9.0.0-rc001" />
+    <PackageReference Include="Umbraco.Cms.Core" Version="9.0.0-rc002" />
+    <PackageReference Include="Umbraco.Cms.Infrastructure" Version="9.0.0-rc002" />
+    <PackageReference Include="Umbraco.Cms.Web.BackOffice" Version="9.0.0-rc002" />
+    <PackageReference Include="Umbraco.Cms.Web.Common" Version="9.0.0-rc002" />
+    <PackageReference Include="Umbraco.Cms.Web.Website" Version="9.0.0-rc002" />
   </ItemGroup>
     <ItemGroup>
         <None Include="../../docs/img/logo.png" Pack="true" PackPath="\" />
diff --git a/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/Backoffice/Dialogs/dialog.html b/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/Backoffice/Dialogs/dialog.html
index 8dab037..fef4f63 100644
--- a/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/Backoffice/Dialogs/dialog.html
+++ b/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/Backoffice/Dialogs/dialog.html
@@ -20,7 +20,7 @@
                     <strong>{{target.name}}</strong> is set as the 404 page for
                     <strong>{{currentNode.name}}</strong> and it's child pages
                 </div>
-                <button class="btn btn-primary" ng-click="nav.hideDialog()">Ok</button>
+                <button class="btn btn-primary" ng-click="vm.close()">Ok</button>
             </div>
 
             <p class="abstract" ng-hide="success">
@@ -49,7 +49,7 @@
                                          section="{{section}}">
                     </umb-tree-search-box>
 
-                    <br/>
+                    <br />
 
                     <umb-tree-search-results ng-if="searchInfo.showSearch"
                                              results="searchInfo.results"
@@ -81,7 +81,7 @@
     </div>
 
     <div class="umb-dialog-footer btn-toolbar umb-btn-toolbar" ng-hide="success || busy">
-        <a class="btn btn-link" ng-click="nav.hideDialog()">
+        <a class="btn btn-link" ng-click="vm.close()">
             <localize key="general_cancel">Cancel</localize>
         </a>
         <button class="btn btn-primary" ng-click="setNotFoundPage()">
diff --git a/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/js/dialog.controller.js b/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/js/dialog.controller.js
index 0af16d6..9217c93 100644
--- a/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/js/dialog.controller.js
+++ b/src/HC.PageNotFoundManager/App_Plugins/HC.PageNotFound/js/dialog.controller.js
@@ -1,4 +1,4 @@
-(function() {
+(function () {
     "use strict";
     function PageNotFoundManagerDialog($scope, pageNotFoundManagerResource, navigationService, userService, entityResource, iconHelper) {
         var node = $scope.currentNode;
@@ -34,7 +34,7 @@
                 const val = parseInt(resp.data);
                 if (!isNaN(val) && angular.isNumber(val) && val > 0) {
                     $scope.pageNotFoundId = val;
-                    entityResource.getById(val, "Document").then(function(item) {
+                    entityResource.getById(val, "Document").then(function (item) {
                         item.icon = iconHelper.convertFromLegacyIcon(item.icon);
                         $scope.pageNotFoundNode = item;
                     });
@@ -43,6 +43,8 @@
                 initNotFound = true;
                 vm.busy = !(initNotFound && initUserDetails);
             });
+
+            vm.close = close;
         }
 
 
@@ -69,7 +71,11 @@
             }
         }
 
-        $scope.hideSearch = function() {
+        function close() {           
+            $scope.nav.hideDialog();
+        }
+
+        $scope.hideSearch = function () {
             $scope.searchInfo.showSearch = false;
             $scope.searchInfo.searchFromId = null;
             $scope.searchInfo.searchFromName = null;
@@ -77,18 +83,18 @@
         };
 
         // method to select a search result
-        $scope.selectResult = function(evt, result) {
+        $scope.selectResult = function (evt, result) {
             result.selected = result.selected === true ? false : true;
             nodeSelectHandler(evt, { event: evt, node: result });
         };
 
         //callback when there are search results
-        $scope.onSearchResults = function(results) {
+        $scope.onSearchResults = function (results) {
             $scope.searchInfo.results = results;
             $scope.searchInfo.showSearch = true;
         };
 
-        $scope.setNotFoundPage = function() {
+        $scope.setNotFoundPage = function () {
 
             vm.busy = true;
             $scope.error = false;
@@ -101,11 +107,11 @@
             if ($scope.target !== undefined && $scope.target !== null)
                 notFoundPageId = $scope.target.id;
 
-            pageNotFoundManagerResource.setNotFoundPage(parentId, notFoundPageId).then(function() {
+            pageNotFoundManagerResource.setNotFoundPage(parentId, notFoundPageId).then(function () {
                 $scope.error = false;
                 $scope.success = true;
                 vm.busy = false;
-            }, function(err) {
+            }, function (err) {
                 $scope.success = false;
                 $scope.error = err;
                 vm.busy = false;
@@ -120,7 +126,7 @@
             }
         }
 
-        $scope.onTreeInit = function() {
+        $scope.onTreeInit = function () {
             $scope.dialogTreeApi.callbacks.treeLoaded(treeLoadedHandler);
             $scope.dialogTreeApi.callbacks.treeNodeSelect(nodeSelectHandler);
             $scope.dialogTreeApi.callbacks.treeNodeExpanded(nodeExpandedHandler);
@@ -130,6 +136,6 @@
     }
 
     angular.module("umbraco").controller("hc.pagenotfound.dialog.controller",
-        ['$scope', 'hc.pageNotFoundManagerResource', 'notificationsService',
-         'userService', 'entityResource', 'iconHelper', PageNotFoundManagerDialog]);
+        ['$scope', 'hc.pageNotFoundManagerResource', 'navigationService',
+            'userService', 'entityResource', 'iconHelper', PageNotFoundManagerDialog]);
 })();
\ No newline at end of file
diff --git a/src/HC.PageNotFoundManager/HC.PageNotFoundManager.csproj b/src/HC.PageNotFoundManager/HC.PageNotFoundManager.csproj
index a3ec7ee..5f137c8 100644
--- a/src/HC.PageNotFoundManager/HC.PageNotFoundManager.csproj
+++ b/src/HC.PageNotFoundManager/HC.PageNotFoundManager.csproj
@@ -3,6 +3,7 @@
     <PropertyGroup>
         <AssemblyVersion>1.0.0</AssemblyVersion>
         <Version>1.0.0</Version>
+        <VersionSuffix>beta-004</VersionSuffix>
         <InformationalVersion>1.0.0</InformationalVersion>
         <TargetFramework>net5.0</TargetFramework>
         <Authors>Nik Rimington</Authors>
@@ -37,4 +38,7 @@
         <None Include="../../docs/img/logo.png" Pack="true" PackPath="\" />
         <None Include="../../LICENSE" Pack="true" PackPath="" />
     </ItemGroup>
+    <ItemGroup>
+        <PackageReference Include="Umbraco.Cms.Core" Version="9.0.0-rc002" />
+    </ItemGroup>
 </Project>