Сортировка TreeMap по значению Integer

TreeMap<String, Integer> wordsMap = new TreeMap<String, Integer>();

                    TreeMap<String, Integer> wordsMap = new TreeMap<>(new Comparator<Integer>()
                    {
                        public int compare(Integer i1, Integer i2)
                        {
                            return i1.compareTo(i2);
                        }
                    }
                    );

public class Sort {
 
    public static void main(String[] args) {
        sort();
    }
 
 
    private static void sort() {
        Map<String, Integer> map = new TreeMap<String, Integer>();
        map.put("A", 3);
        map.put("B", 4);
        map.put("C", 1);
        map = sortByValues(map);
 
        System.out.println(map);
    }
 
    public static <K, V extends Comparable<V>> Map<K, V> sortByValues(final Map<K, V> map) {
        Comparator<K> valueComparator = new Comparator<K>() {
            public int compare(K k1, K k2) {
                return map.get(k1).compareTo(map.get(k2));
            }
        };
        Map<K, V> sortedByValues = new TreeMap<K, V>(valueComparator);
        sortedByValues.putAll(map);
        return sortedByValues;
    }
}

                    List list = new ArrayList(wordsMap.entrySet());
                    Collections.sort(list, new Comparator<Map.Entry<String, Integer>>()
                    {
                       @Override
                       public int compare(Map.Entry<String, Integer> a, Map.Entry<String, Integer> b)
                       {
                           return b.getValue() - a.getValue();
                       }
                    });

    public static void CSVTable(String str, int wordsInTheText)
    {
        String key = "";
        String value = "";
        int i = 0;
 
        while (str.charAt(i) != '=')
        {
            key += str.charAt(i);
            i++;
        }
        i++;
        while (i < str.length() && (Character.isLetterOrDigit(str.charAt(i))))
        {
            value += str.charAt(i);
            i++;
        }
        System.out.print(key);
        System.out.print("; ");
        System.out.print(value);
        System.out.print("; ");
        System.out.print((double) Integer.parseInt(value) / wordsInTheText * 100 + " %;");
        System.out.println("");
    }