--- a/FrmNewUnitFromTree.cs	Tue Aug 16 21:01:45 2011 +0100
+++ b/FrmNewUnitFromTree.cs	Sat Aug 20 16:12:15 2011 +0100
@@ -19,7 +19,7 @@
             WarFoundryCore.ArmyChanged += new ArmyChangedDelegate(ArmyChanged);
         }
 
-        private void ArmyChanged(Army newArmy, Army oldArmy)
+        private void ArmyChanged(Army oldArmy, Army newArmy)
         {
             unitTypeTree.Nodes.Clear();
 
@@ -49,6 +49,27 @@
                 unitTypeTree.Nodes.Add(catTemp);
                 catTemp.ExpandAll();
             }
+
+            unitTypeTree.Show();
+        }
+
+        private void unitTypeTree_DoubleClick(object sender, EventArgs e)
+        {
+            TreeNode selected = unitTypeTree.SelectedNode;
+            object tagData = selected.Tag;
+
+            if (tagData is UnitType)
+            {
+                UnitType unit = (UnitType)tagData;
+                Category cat = (Category)selected.Parent.Tag;
+                MdiParent.CreateUnitInCategory(unit, cat);
+            }
+        }
+
+        public new FrmMain MdiParent
+        {
+            get { return (FrmMain)base.MdiParent; }
+            set { base.MdiParent = value; }
         }
     }
 }