From 4ad804252205ef5b0e6a0e60d4302a087917eceb Mon Sep 17 00:00:00 2001
From: mherbaghinyan <mbaghinyan@gmail.com>
Date: Sat, 5 May 2018 10:26:51 +0400
Subject: [PATCH] test the results

---
 .../designpatterns/composite/CompositeDemo.java        | 10 +++++++++-
 .../{GeneralDepartment.java => HeadDepartment.java}    |  4 ++--
 2 files changed, 11 insertions(+), 3 deletions(-)
 rename core-java/src/main/java/com/baeldung/designpatterns/composite/{GeneralDepartment.java => HeadDepartment.java} (86%)

diff --git a/core-java/src/main/java/com/baeldung/designpatterns/composite/CompositeDemo.java b/core-java/src/main/java/com/baeldung/designpatterns/composite/CompositeDemo.java
index 83024b7d21..9537dd0d2b 100644
--- a/core-java/src/main/java/com/baeldung/designpatterns/composite/CompositeDemo.java
+++ b/core-java/src/main/java/com/baeldung/designpatterns/composite/CompositeDemo.java
@@ -6,6 +6,14 @@ package com.baeldung.designpatterns.composite;
 public class CompositeDemo {
 
     public static void main(String args[]) {
-        Department salesDepartment = new SalesDepartment();
+        Department salesDepartment = new SalesDepartment(1, "Sales department");
+        Department financialDepartment = new FinancialDepartment(2, "Financial department");
+
+        HeadDepartment headDepartment = new HeadDepartment(3, "Head department");
+
+        headDepartment.addDepartMent(salesDepartment);
+        headDepartment.addDepartMent(financialDepartment);
+
+        headDepartment.printDepartmentName();
     }
 }
diff --git a/core-java/src/main/java/com/baeldung/designpatterns/composite/GeneralDepartment.java b/core-java/src/main/java/com/baeldung/designpatterns/composite/HeadDepartment.java
similarity index 86%
rename from core-java/src/main/java/com/baeldung/designpatterns/composite/GeneralDepartment.java
rename to core-java/src/main/java/com/baeldung/designpatterns/composite/HeadDepartment.java
index 15c5bc9494..8854fce5b3 100644
--- a/core-java/src/main/java/com/baeldung/designpatterns/composite/GeneralDepartment.java
+++ b/core-java/src/main/java/com/baeldung/designpatterns/composite/HeadDepartment.java
@@ -6,14 +6,14 @@ import java.util.List;
 /**
  * Created by Gebruiker on 5/1/2018.
  */
-public class GeneralDepartment implements Department {
+public class HeadDepartment implements Department {
 
     private Integer id;
     private String name;
 
     private List<Department> childDepartments;
 
-    public GeneralDepartment(Integer id, String name) {
+    public HeadDepartment(Integer id, String name) {
         this.id = id;
         this.name = name;
         this.childDepartments = new ArrayList<Department>();