diff --git a/src/main/java/com/thealgorithms/maths/FindMax.java b/src/main/java/com/thealgorithms/maths/FindMax.java
index 0ff2bdd191ac..ba5998b89e68 100644
--- a/src/main/java/com/thealgorithms/maths/FindMax.java
+++ b/src/main/java/com/thealgorithms/maths/FindMax.java
@@ -4,23 +4,14 @@ public final class FindMax {
     private FindMax() {
     }
 
-    /**
-     * @brief finds the maximum value stored in the input array
-     *
-     * @param array the input array
-     * @exception IllegalArgumentException input array is empty
-     * @return the maximum value stored in the input array
-     */
     public static int findMax(final int[] array) {
-        int n = array.length;
-        if (n == 0) {
+        if (array.length == 0) {
             throw new IllegalArgumentException("Array must be non-empty.");
         }
+
         int max = array[0];
-        for (int i = 1; i < n; i++) {
-            if (array[i] > max) {
-                max = array[i];
-            }
+        for (int i = 1; i < array.length; i++) {
+            max = Math.max(array[i], max);
         }
         return max;
     }