feat: Implement Assignment 3 functionality

1 files changed, 4 insertions, 5 deletions

diff --git a/MorgSimulator/Reader/MorgReader.cs b/MorgSimulator/Reader/MorgReader.cs
index 2bf9a30..56c229b 100644
--- a/MorgSimulator/Reader/MorgReader.cs
+++ b/MorgSimulator/Reader/MorgReader.cs
@@ -1,13 +1,12 @@
 #nullable enable
 using MorgSimulator.Factory;
+using MorgSimulator.Framework;
 
 namespace MorgSimulator.Reader
 {
-    public class MorgReader(Reader reader, IMorgFactory factory) : ReaderDecorator(reader)
+    public class MorgReader(Reader reader, IEntityFactory<Morg> factory) : EntityReader<Morg>(reader, factory)
     {
-        private readonly IMorgFactory _factory = factory;
-
-        public Morg? ReadMorg(int id)
+        public override Morg? ReadEntity(int id)
         {
             if (_reader is not CSVReader csvReader) return null;
 
@@ -25,7 +24,7 @@ namespace MorgSimulator.Reader
                 !int.TryParse(yString, out int y))
                 return null;
 
-            return _factory.CreateMorg(id, type, x, y, movement, feeding);
+            return _factory.CreateEntity(id, type, x, y, movement, feeding);
         }
     }
 }