diff --git a/src/paramak/assemblies/assembly.py b/src/paramak/assemblies/assembly.py
index f7bf0921..5e65b920 100644
--- a/src/paramak/assemblies/assembly.py
+++ b/src/paramak/assemblies/assembly.py
@@ -28,18 +28,3 @@ def names(self):
         for part in self:
             names.append(part[1].split('/')[-1])
         return names
-
-
-
-
-# testing
-# sphere1 = cq.Workplane().moveTo(2, 2).sphere(1)
-# box1 = cq.Workplane().box(1, 1, 1)
-# assembly = Assembly()
-# assembly.add(box1, name="box1", color=cq.Color(0.5, 0.5, 0.5))
-# assembly.add(sphere1, name="sphere")
-
-
-# assembly2 = assembly.remove('sphere')
-# assembly3 = assembly.remove('box1')
-# assembly4 = assembly.remove('bosdfsdf')
\ No newline at end of file
diff --git a/tests/test_assemblies/test_assembly.py b/tests/test_assemblies/test_assembly.py
new file mode 100644
index 00000000..b1581d70
--- /dev/null
+++ b/tests/test_assemblies/test_assembly.py
@@ -0,0 +1,20 @@
+import cadquery as cq
+from paramak.assemblies.assembly import Assembly
+
+def test_remove_and_names():
+
+    sphere1 = cq.Workplane().moveTo(2, 2).sphere(1)
+    box1 = cq.Workplane().box(1, 1, 1)
+    assembly = Assembly()
+    assembly.add(box1, name="box1", color=cq.Color(0.5, 0.5, 0.5))
+    assembly.add(sphere1, name="sphere")
+
+
+    assembly2 = assembly.remove('sphere')
+    assembly3 = assembly.remove('box1')
+    assembly4 = assembly.remove('bosdfsdf')
+
+    assert assembly.names() == ['box1', 'sphere']
+    assert assembly2.names() == ['box1']
+    assert assembly3.names() == ['sphere']
+    assert assembly4.names() == ['box1', 'sphere']