diff --git a/MainScript.m b/MainScript.m
index 355acd7..4bdee22 100644
--- a/MainScript.m
+++ b/MainScript.m
@@ -9,9 +9,9 @@
     
     
     %% error message for when you try to run some of the functions without loading system.
-    if sId == -1 && (choice == 3)
+    if sId == -1 && (choice == 2)
         disp('Please choose a type of Lindenmeyer system first: ');
-        roadblock= imread('roadblock.jpg');
+        roadblock= imread('gandalf.jpg');
         image(roadblock);
     else
              
@@ -26,11 +26,15 @@
                 
                 %Sub system menu
                 if sId == 1
+                    System = 'koch'
                     disp('You have chosen the Koch curve!')
-                    iterations = input('Please enter amount of iterations: ')
+                    N = input('Please enter amount of iterations: ')
+                    LindenMayerString=LindIter(System,N)
                 elseif sId == 2
+                    System = 'sierpinski'
                     disp('You have chosen the Sierpinski triangle!')
-                    iterations = input('Please enter amount of iterations: ')
+                    N = input('Please enter amount of iterations: ')
+                    LindenMayerString=LindIter(System,N)
                 end
                
                 if sId == -1
diff --git a/gandalf.jpg b/gandalf.jpg
new file mode 100644
index 0000000..817e6f6
Binary files /dev/null and b/gandalf.jpg differ