diff --git a/ProjectElectricLocomotive/LocomotiveGenericCollection.java b/ProjectElectricLocomotive/LocomotiveGenericCollection.java index 25f45b4..e58fc66 100644 --- a/ProjectElectricLocomotive/LocomotiveGenericCollection.java +++ b/ProjectElectricLocomotive/LocomotiveGenericCollection.java @@ -91,4 +91,8 @@ public class LocomotiveGenericCollection collection; if (_locomotiveStorage.containsKey(key)){ collection = _locomotiveStorage.get(key); -// collection.clear(); // what is the wtf??? + collection.clear(); } else collection = new LocomotiveGenericCollection<>(_pictureWidth, _pictureHeight); diff --git a/ProjectElectricLocomotive/SetGeneric.java b/ProjectElectricLocomotive/SetGeneric.java index f731887..062252a 100644 --- a/ProjectElectricLocomotive/SetGeneric.java +++ b/ProjectElectricLocomotive/SetGeneric.java @@ -66,4 +66,7 @@ public class SetGeneric{ return _places; } + public void clear() { + _places.clear(); + } } \ No newline at end of file