memoizeit

diff --git a/src/main/java/br/ufrgs/inf/prosoft/memoizeit/Main.java b/src/main/java/br/ufrgs/inf/prosoft/memoizeit/Main.java
index b43ccaa..913f5a2 100644
--- a/src/main/java/br/ufrgs/inf/prosoft/memoizeit/Main.java
+++ b/src/main/java/br/ufrgs/inf/prosoft/memoizeit/Main.java
@@ -40,11 +40,7 @@ public class Main {
                 return new String[]{arg, ""};
             }
             return new String[]{arg.substring(0, indexOf), arg.substring(indexOf + 1)};
-        }).collect(Collectors.toMap(array -> {
-            return array[0];
-        }, array -> {
-            return array[1];
-        }));
+        }).collect(Collectors.toMap(array -> array[0], array -> array[1]));
 
         String tracePath = arguments.get("trace");
         if (tracePath == null) {

diff --git a/src/main/java/br/ufrgs/inf/prosoft/memoizeit/MemoizeIt.java b/src/main/java/br/ufrgs/inf/prosoft/memoizeit/MemoizeIt.java
index a3bfdbe..3d48930 100644
--- a/src/main/java/br/ufrgs/inf/prosoft/memoizeit/MemoizeIt.java
+++ b/src/main/java/br/ufrgs/inf/prosoft/memoizeit/MemoizeIt.java
@@ -8,11 +8,8 @@ package br.ufrgs.inf.prosoft.memoizeit;
 import br.ufrgs.inf.prosoft.cache.CachePerformance;
 import br.ufrgs.inf.prosoft.memoizeit.graph.Graph;
 import br.ufrgs.inf.prosoft.memoizeit.graph.Node;
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.Collections;
-import java.util.List;
-import java.util.Map;
+
+import java.util.*;
 import java.util.function.Consumer;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -139,10 +136,8 @@ public class MemoizeIt {
 
     private void refineCandidates(boolean iteratively) {
         LOGGER.log(Level.INFO, "Refining {0} candidates", this.methods.size());
-        this.methods.sort((m1, m2) -> Integer.compare(m1.getOccurrencesSize(), m2.getOccurrencesSize()));
-        this.methods.removeIf(method -> {
-            return iteratively ? refineCandidateIteratively(method) : refineCandidateExhaustively(method);
-        });
+        this.methods.sort(Comparator.comparingInt(Method::getOccurrencesSize));
+        this.methods.removeIf(method -> iteratively ? refineCandidateIteratively(method) : refineCandidateExhaustively(method));
     }
 
     private boolean refineCandidateExhaustively(Method method) {