diff --git a/core-java-modules/core-java-collections-maps-2/src/main/java/com/baeldung/map/mapmax/MapMax.java b/core-java-modules/core-java-collections-maps-2/src/main/java/com/baeldung/map/mapmax/MapMax.java
index 99b57b398a..11443865c7 100644
--- a/core-java-modules/core-java-collections-maps-2/src/main/java/com/baeldung/map/mapmax/MapMax.java
+++ b/core-java-modules/core-java-collections-maps-2/src/main/java/com/baeldung/map/mapmax/MapMax.java
@@ -1,7 +1,6 @@
 package com.baeldung.map.mapmax;
 
 import java.util.Collections;
-import java.util.Comparator;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Map.Entry;
@@ -26,12 +25,7 @@ public class MapMax {
 
     public <K, V extends Comparable<V>> V maxUsingCollectionsMax(Map<K, V> map) {
 
-        Entry<K, V> maxEntry = Collections.max(map.entrySet(), new Comparator<Entry<K, V>>() {
-            public int compare(Entry<K, V> e1, Entry<K, V> e2) {
-                return e1.getValue()
-                    .compareTo(e2.getValue());
-            }
-        });
+        Entry<K, V> maxEntry = Collections.max(map.entrySet(), Entry.comparingByValue());
 
         return maxEntry.getValue();
     }
@@ -72,7 +66,7 @@ public class MapMax {
 
     public static void main(String[] args) {
 
-        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
+        Map<Integer, Integer> map = new HashMap<>();
 
         map.put(1, 3);
         map.put(2, 4);