diff --git a/src/storage/jani/ModelType.cpp b/src/storage/jani/ModelType.cpp
index 243d69515..09a28f7e0 100644
--- a/src/storage/jani/ModelType.cpp
+++ b/src/storage/jani/ModelType.cpp
@@ -20,9 +20,37 @@ namespace storm {
                 case ModelType::MA:
                     stream << "ma";
                     break;
+                case ModelType::PTA:
+                    stream << "pta";
+                    break;
+                case ModelType::STA:
+                    stream << "sta";
+                    break;
             }
             return stream;
         }
+
+        ModelType getModelType(std::string const& input) {
+            if(input == "dtmc") {
+                return ModelType::DTMC;
+            }
+            if(input == "ctmc") {
+                return ModelType::CTMC;
+            }
+            if(input == "mdp") {
+                return ModelType::MDP;
+            }
+            if(input == "ma") {
+                return ModelType::MA;
+            }
+            if(input == "pta") {
+                return ModelType::PTA;
+            }
+            if(input == "sta") {
+                return ModelType::STA;
+            }
+            return ModelType::UNDEFINED;
+        }
         
     }
 }
diff --git a/src/storage/jani/ModelType.h b/src/storage/jani/ModelType.h
index 47568355e..80a78ad49 100644
--- a/src/storage/jani/ModelType.h
+++ b/src/storage/jani/ModelType.h
@@ -5,8 +5,9 @@
 namespace storm {
     namespace jani {
         
-        enum class ModelType {UNDEFINED = 0, DTMC = 1, CTMC = 2, MDP = 3, MA = 4};
-     
+        enum class ModelType {UNDEFINED = 0, DTMC = 1, CTMC = 2, MDP = 3, MA = 4, PTA = 5, STA = 6};
+
+        ModelType getModelType(std::string const& input);
         std::ostream& operator<<(std::ostream& stream, ModelType const& type);
         
     }