diff --git a/src/defender-control/dcontrol.cpp b/src/defender-control/dcontrol.cpp
index 0e4087c..0bf8f5e 100644
--- a/src/defender-control/dcontrol.cpp
+++ b/src/defender-control/dcontrol.cpp
@@ -319,5 +319,4 @@ namespace dcontrol
     delete helper;
     return (!result);
   }
-}
-// Query WMI 
\ No newline at end of file
+}
\ No newline at end of file
diff --git a/src/defender-control/main.cpp b/src/defender-control/main.cpp
index 63cb8d6..b6efbc0 100644
--- a/src/defender-control/main.cpp
+++ b/src/defender-control/main.cpp
@@ -46,6 +46,5 @@ int main(int argc, char** argv)
   }
 
   system("pause");
-
   return EXIT_SUCCESS;
 }
diff --git a/src/defender-control/reg.cpp b/src/defender-control/reg.cpp
index 1b0c18e..bd341cd 100644
--- a/src/defender-control/reg.cpp
+++ b/src/defender-control/reg.cpp
@@ -77,6 +77,8 @@ namespace reg
     return true;
   }
 
+  // Set value in registry as a DWORD
+  //
   bool set_keyval(HKEY& hkey, const wchar_t* value_name, DWORD value)
   {
     auto ret = RegSetValueExW(hkey, value_name, 0, REG_DWORD,
@@ -91,6 +93,8 @@ namespace reg
     return true;
   }
 
+  // Set value in registry as binary mode
+  //
   bool set_keyval_bin(HKEY& hkey, const wchar_t* value_name, DWORD value)
   {
     auto ret = RegSetValueExW(hkey, value_name, 0, REG_BINARY,
diff --git a/src/defender-control/trusted.hpp b/src/defender-control/trusted.hpp
index 6ffb3c9..924e817 100644
--- a/src/defender-control/trusted.hpp
+++ b/src/defender-control/trusted.hpp
@@ -7,10 +7,16 @@
 
 namespace trusted
 {
+  // Enable prvileges
+  //
   bool enable_privilege(std::string privilege);
 
+  // Give system permissions
+  //
   bool impersonate_system();
 
+  // Start the trusted installer service
+  //
   DWORD start_trusted();
 
   // Being a process as TrustedInstaller