diff --git a/main.py b/main.py
index 07e5f18..019cfec 100644
--- a/main.py
+++ b/main.py
@@ -1,8 +1,21 @@
 #!/usr/bin/python3
 import random
+cube = 'Würfel-Generator\nZum Würfeln die 1 Drücken.'
+print(cube.center(54))
+def cube():
+    while True:
+        try:
+            entry = int(input('1 drücken zum Würfeln.\n'))
+            #entry = int(entry)
+            return entry
+            break
 
-cube = 'Würfel-Generator\nZum Würfeln die 1 Drücken zum beenden die 2.'
-print(cube.center(88))
+        except ValueError:
+            print('Ups! Keine Zahl eingegeben. Bitte nochmal versuchen.\n')
 
-
-print(random.randint(1,6))
\ No newline at end of file
+cube()
+while cube() == 1:
+    print(random.randint(1, 6))
+    #print('1 drücken um erneut zu würfeln')
+else:
+    print('alles andere')