diff --git a/GoldWars/GoldWars/Unit/Unit.swift b/GoldWars/GoldWars/Unit/Unit.swift
index f01eba3..a4fa851 100644
--- a/GoldWars/GoldWars/Unit/Unit.swift
+++ b/GoldWars/GoldWars/Unit/Unit.swift
@@ -6,8 +6,8 @@
 //  Copyright © 2020 SP2. All rights reserved.
 //
 
-struct Unit{
-    var strength: Float
-    var health: Float
-    var defence: Float
+enum Unit{
+    case General
+    case Defence
+    case Attack
 }