diff --git a/Main.java b/Main.java
index a112805..880bb57 100644
--- a/Main.java
+++ b/Main.java
@@ -27,6 +27,13 @@ public static void rearrange(int arr[], int n)
                 min_idx++; 
             } 
         } 
+      
+      # A for loop can be added for better optimization
+        
+        for (int i =0 ; i<n; i++)
+        {
+          arr[i] = arr[i] + 1;
+        }
   
         for (int i = 0; i < n; i++) 
             arr[i] = arr[i] / max_elem;