diff --git a/main.py b/main.py
old mode 100644
new mode 100755
index 9d4cee4..3285609
--- a/main.py
+++ b/main.py
@@ -1,18 +1,26 @@
 #!/usr/bin/python3
 import random
-cube = 'Würfel-Generator\nZum Würfeln die 1 Drücken.'
+import os
+
+cube = 'Würfel-Generator\n'
 print(cube.center(54))
 def cube():
     while True:
         try:
-            entry = int(input('1 drücken zum Würfeln.\n'))
-            #entry = int(entry)
+            entry = int(input('1 drücken zum Würfeln.\n2 drücken zum beenden.\n'))
+
             if entry == 1:
+                os.system("clear")
                 print(random.randint(1, 6))
+            elif entry == 0:
+                print('Ups! Da wurde nicht 1 oder 2 gedrückt. Bitte nochmal versuchen.\n')
+            elif entry == 2:
+                print('Danke für\'s würfeln.\nBis Bald!')
+                exit()
             elif entry > 1:
-                print('Ups! Da wurde nicht die 1 gedrückt. Bitte nochmal versuchen.\n')
+                print('Ups! Da wurde nicht 1 oder 2 gedrückt. Bitte nochmal versuchen.\n')
         except ValueError:
-            print('Ups! Da wurde nicht die 1 gedrückt. Bitte nochmal versuchen.\n')
+            print('Ups! Da wurde nicht 1 oder 2 gedrückt. Bitte nochmal versuchen.\n')
 
 cube()