ROT working completely

crypttool.py

@@ -84,6 +84,7 @@ def fixedxor(string1 = '', string2 = ''):
 
 def rot(inputString, amount):
     outputString = '' 
+    amount = int(amount)
     for char in inputString:
         resultChar = ''
         if char.isupper():
@@ -237,11 +238,11 @@ def base64prompt():
 def rotPrompt():
     choice = input('What kind of ROT do you want to perform? 1-25, or all: ')
     userInput = input('Please insert a string: ')
-    if type(choice) is types.IntType:
-        print(rot(userInput, choice))
-    else:
+    if choice == 'all':
       for i in range(0, 26):
         print(rot(userInput, i))
+    else:
+      print(rot(userInput, choice))
     
 def translatePrompt():
     print('1: Binary')