added main and few extensions

ExtensionName.py

@@ -3,12 +3,13 @@ def foldername(extension):
         return None
     else:
         switcher = {
-            "exe" : "Software/Applications", 
-            "txt" : "Notes/Text", 
+            "exe"  : "Software", 
+            "txt"  : "Notes-Text", 
             "pdf"  : "PDFs", 
             "c"    : "C programs", 
             "py"   : "Python files",
             "java" : "Java programs",
+            "class": "Java programs", 
             "cpp"  : "Cpp programs", 
             "jpg"  : "Images", 
             "png"  : "Images",  

FileOrganizeAutomation.py

@@ -13,9 +13,6 @@ paths = {
     "D Drive"   : "D:/"
 }
 
-print("\n\n1:Organize Files\n2:Backup To Drive\n3:Rename Unordered Files")
-choice = int(input("\nEnter choice:"))
-
 def organizeFiles():
     print("\n\nFolders...\n")
     for key in paths.keys():
@@ -64,6 +61,10 @@ def organizeFiles():
     return 0
 
 
+def main():
+    print("\n\n1:Organize Files\n2:Backup To Drive\n3:Rename Unordered Files")
+    choice = int(input("\nEnter choice:"))
+
     if(choice == 1):
         organizeFiles()
     elif(choice == 2):
@@ -74,5 +75,9 @@ else:
         print("\nInvalid Choice")
 
 
+if __name__ == "__main__":
+    main()
+
+