Update StalinSort.java

Author: anant-jain01

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

@@ -18,15 +18,20 @@ public <T extends Comparable<T>> T[] sort(T[] array) {
             }
         }
 
+        // Create a result array with sorted elements
         T[] result = (T[]) java.lang.reflect.Array.newInstance(array.getClass().getComponentType(), 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));
     }