Update StalinSort.java

Author: anant-jain01

src/main/java/com/thealgorithms/sorts/StalinSort.java

@@ -16,17 +16,26 @@ public <T extends Comparable<T>> T[] sort(T[] array) {
             }
         }
 
+        // Create a result array with sorted elements
         T[] result = (T[]) new Comparable[currentIndex + 1];
         System.arraycopy(array, 0, result, 0, currentIndex + 1);
+
         return result;
     }
 
+    // Driver Program
     public static void main(String[] args) {
+        // Integer Input
         Integer[] integers = {4, 23, 6, 78, 1, 54, 231, 9, 12};
+
         StalinSort stalinSort = new StalinSort();
+
+        // print a sorted array
         SortUtils.print(stalinSort.sort(integers));
-        
+
+        // String Input
         String[] strings = {"c", "a", "e", "b", "d"};
+
         SortUtils.print(stalinSort.sort(strings));
     }
 }