diff --git a/fltk/util/cluster/client.py b/fltk/util/cluster/client.py
index 62af8306..853d8805 100644
--- a/fltk/util/cluster/client.py
+++ b/fltk/util/cluster/client.py
@@ -169,6 +169,9 @@ class ClusterManager(abc.ABC):
     Object with basic monitoring functionality. This shows how the information of different Pods in a cluster can be
     requested and parsed. Currently, it mainly exists to start the ResourceWatchDog, which now only keeps track of the
     amount of resources requested/used in the cluster.
+
+    This object is provided to the Orchestrator to allow for keeping track of available resources to allow for
+    scheduling of training tasks for extending an Orchestrator for experiments..
     """
 
     def __init__(self):
diff --git a/fltk/util/config/definitions/net.py b/fltk/util/config/definitions/net.py
index b6f18ebe..1014654c 100644
--- a/fltk/util/config/definitions/net.py
+++ b/fltk/util/config/definitions/net.py
@@ -4,26 +4,25 @@
 
 @unique
 class Nets(Enum):
-    """ """
-    cifar100_resnet = "Cifar100ResNet"
-    cifar100_vgg = "Cifar100VGG"
-    cifar10_cnn = "Cifar10CNN"
-    cifar10_resnet = "Cifar10ResNet"
-    fashion_mnist_cnn = "FashionMNISTCNN"
-    fashion_mnist_resnet = "FashionMNISTResNet"
+    cifar100_resnet = 'Cifar100ResNet'
+    cifar100_vgg = 'Cifar100VGG'
+    cifar10_cnn = 'Cifar10CNN'
+    cifar10_resnet = 'Cifar10ResNet'
+    fashion_mnist_cnn = 'FashionMNISTCNN'
+    fashion_mnist_resnet = 'FashionMNISTResNet'
     mnist_cnn = 'MNISTCNN'
 
     @classmethod
-    def _missing_name_(cls, name: str) -> Optional["Nets"]:
-        """Helper function to get lower/higher-case configured network, to allow for case-insensitive lookup.
+    def _missing_name_(cls, name: str) -> 'Dataset':
+        '''Helper function in case name could not be looked up (to support older configurations).
 
         Args:
-          name (str): Name of network to lookup in case of missing lookup.
+            name (str): Name of Type to be looked up.
 
         Returns:
-            Optional[Nets]: Of name is not part of defined networks, else reference to class implementing network.
+            Dataset: Corresponding Enum instance, if name is recognized from lower case.
 
-        """
+        '''
         for member in cls:
             if member.name.lower() == name.lower():
-                return member
+                return member
\ No newline at end of file
diff --git a/fltk/util/task/arrival_task.py b/fltk/util/task/arrival_task.py
index d23041a5..f0ca0d57 100644
--- a/fltk/util/task/arrival_task.py
+++ b/fltk/util/task/arrival_task.py
@@ -29,7 +29,7 @@ class ArrivalTask(abc.ABC):
     loss_function: str = field(compare=False)
     seed: int = field(compare=False)
     replication: int = field(compare=False)
-    type_map: "Optional[FrozenOrderedDict[str, int]]"
+    type_map: "Optional[FrozenOrderedDict[str, int]]" = field(compare=False)
     system_parameters: SystemParameters = field(compare=False)
     hyper_parameters: HyperParameters = field(compare=False)
     learning_parameters: LearningParameters = field(compare=False)