diff --git a/AIStudio.Wpf.DiagramApp/ViewModels/ToolBoxViewModel.cs b/AIStudio.Wpf.DiagramApp/ViewModels/ToolBoxViewModel.cs
index 6aab3aa..4c83168 100644
--- a/AIStudio.Wpf.DiagramApp/ViewModels/ToolBoxViewModel.cs
+++ b/AIStudio.Wpf.DiagramApp/ViewModels/ToolBoxViewModel.cs
@@ -49,11 +49,6 @@ namespace AIStudio.Wpf.DiagramApp.ViewModels
             }
         }
 
-        public ObservableCollection<ToolBoxCategory> TotalToolBoxCategory
-        {
-            get; set;
-        }
-
         public ToolBoxCategory MyToolBoxCategory
         {
             get
@@ -359,53 +354,6 @@ namespace AIStudio.Wpf.DiagramApp.ViewModels
         }
     }
 
-    public class ToolBoxCategory : BindableBase
-    {
-        public string Header
-        {
-            get; set;
-        }
-
-        private bool _isExpanded;
-        public bool IsExpanded
-        {
-            get
-            {
-                return _isExpanded;
-            }
-            set
-            {
-                SetProperty(ref _isExpanded, value);
-            }
-        }
-
-        private bool _isChecked = true;
-        public bool IsChecked
-        {
-            get
-            {
-                return _isChecked;
-            }
-            set
-            {
-                SetProperty(ref _isChecked, value);
-            }
-        }
-
-        private ObservableCollection<ToolBoxData> _toolBoxItems;
-        public ObservableCollection<ToolBoxData> ToolBoxItems
-        {
-            get
-            {
-                return _toolBoxItems;
-            }
-            set
-            {
-                SetProperty(ref _toolBoxItems, value);
-            }
-        }
-    }
-
     public class MineToolBoxCategory : ToolBoxCategory
     {
 
diff --git a/AIStudio.Wpf.DiagramDesigner/Models/ToolBoxData.cs b/AIStudio.Wpf.DiagramDesigner/Models/ToolBoxData.cs
index 6fafc6c..eefedc0 100644
--- a/AIStudio.Wpf.DiagramDesigner/Models/ToolBoxData.cs
+++ b/AIStudio.Wpf.DiagramDesigner/Models/ToolBoxData.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Collections.ObjectModel;
 using System.Linq;
 using System.Text;
 using System.Windows;
@@ -63,4 +64,51 @@ namespace AIStudio.Wpf.DiagramDesigner
             this.Description = description;
         }
     }
+
+    public class ToolBoxCategory : BindableBase
+    {
+        public string Header
+        {
+            get; set;
+        }
+
+        private bool _isExpanded;
+        public bool IsExpanded
+        {
+            get
+            {
+                return _isExpanded;
+            }
+            set
+            {
+                SetProperty(ref _isExpanded, value);
+            }
+        }
+
+        private bool _isChecked = true;
+        public bool IsChecked
+        {
+            get
+            {
+                return _isChecked;
+            }
+            set
+            {
+                SetProperty(ref _isChecked, value);
+            }
+        }
+
+        private ObservableCollection<ToolBoxData> _toolBoxItems;
+        public ObservableCollection<ToolBoxData> ToolBoxItems
+        {
+            get
+            {
+                return _toolBoxItems;
+            }
+            set
+            {
+                SetProperty(ref _toolBoxItems, value);
+            }
+        }
+    }
 }