Сортировка Map по количеству элементов в ArrayList (value)

import java.util.*;
import java.util.Map.Entry;
 
public class SortingByValDemo {
 
        
    public static void main(String[] args) {
        TreeMap<String,ArrayList<String>> m = new TreeMap<String,ArrayList<String>>();
        
        ArrayList<String> a1 = new ArrayList<String>();
        a1.add("ah");
        a1.add("fuh");
        ArrayList<String> a2 = new ArrayList<String>();
        a2.add("Bling");
        a2.add("Fuck");
        a2.add("Tick");
        a2.add("Buuuu");
        ArrayList<String> a3 = new ArrayList<String>();
        a3.add("111");
        ArrayList<String> a4 = new ArrayList<String>();
        a4.add("nononono");
        m.put("a2", a2);
        m.put("a1", a1);
        m.put("a3", a3);
        m.put("a4", a4);
        
        SortedSet<Map.Entry<String, ArrayList<String>>> sortedset = new TreeSet<Map.Entry<String, ArrayList<String>>>(
                new Comparator<Entry<String,ArrayList<String>>>() {
 
                    @Override
                    public int compare(Entry<String, ArrayList<String>> arg0,
                            Entry<String, ArrayList<String>> arg1) {
                        if (arg0.getValue().size() > arg1.getValue().size())
                            return 1;
                        else if (arg0.getValue().size() < arg1.getValue().size())
                            return -1;
                        else
                            return 1;
                    }
                }
            );
        sortedset.addAll(m.entrySet());
        for (Map.Entry<String, ArrayList<String>> em : sortedset){
            System.out.println(
                    em.getKey()+" : "+em.getValue()+"\n"+"========="+"\n"
                    
            );
        }
        
    }
 
}