added main and few extensions

ExtensionName.py

@@ -3,16 +3,17 @@ 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", 
+            "java" : "Java programs",
+            "class": "Java programs", 
             "cpp"  : "Cpp programs", 
             "jpg"  : "Images", 
             "png"  : "Images",  
-            "jpeg": "Images", 
+            "jpeg" : "Images", 
             "raw"  : "Images",
             "mp3"  : "Music", 
             "mp4"  : "Videos", 

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,15 +61,23 @@ def organizeFiles():
     return 0
 
 
-if(choice == 1):
+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):
+    elif(choice == 2):
         print("Backup feature not available yet")
-elif(choice == 3):
+    elif(choice == 3):
         print("Feature to be added soon")
-else:
+    else:
         print("\nInvalid Choice")
 
 
+if __name__ == "__main__":
+    main()
+
+