diff --git a/Disco.Plugins.ADCompare.csproj b/Disco.Plugins.ADCompare.csproj
index 8045e9e..391d117 100644
--- a/Disco.Plugins.ADCompare.csproj
+++ b/Disco.Plugins.ADCompare.csproj
@@ -32,22 +32,16 @@
   <ItemGroup>
     <Reference Include="System" />
     <Reference Include="System.Core" />
-    <Reference Include="System.Drawing" />
+    <Reference Include="System.Data" />
+    <Reference Include="System.Data.Entity" />
     <Reference Include="System.Web" />
     <Reference Include="System.Web.Mvc, Version=5.2.7.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35" />
     <Reference Include="System.ComponentModel.DataAnnotations" />
-    <Reference Include="System.Data" />
-    <Reference Include="System.Data.Entity" />
     <Reference Include="Newtonsoft.Json">
       <HintPath>..\packages\Newtonsoft.Json.13.0.1\lib\net45\Newtonsoft.Json.dll</HintPath>
     </Reference>
   </ItemGroup>
   <ItemGroup>
-    <!-- 
-      Reference the Disco assemblies from the main solution.
-      Adjust these paths to match your build output or NuGet package location.
-      If building within the Disco solution, use ProjectReference instead.
-    -->
     <Reference Include="Disco.Models">
       <HintPath>..\Disco.Models\bin\Debug\Disco.Models.dll</HintPath>
       <Private>False</Private>
@@ -63,9 +57,9 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="ADComparePlugin.cs" />
-    <Compile Include="Features\ADCompareDetailsProvider.cs" />
-    <Compile Include="Features\ADCompareService.cs" />
-    <Compile Include="Models\UserComparisonResult.cs" />
+    <Compile Include="ConfigurationHandler\ADCompareConfigurationHandler.cs" />
+    <Compile Include="Features\DeviceCompareService.cs" />
+    <Compile Include="Models\DeviceComparisonResult.cs" />
     <Compile Include="WebHandler\ADCompareWebHandler.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
   </ItemGroup>