Дан массив. Напиши код на джава, чтобы найти количество элементов массива, значение которых меньше среднего арифметического, взятого от всех элементов массива

public static int countElementsBelowAverage(int[] arr) {
    int sum = 0;
    for (int i = 0; i < arr.length; i++) {
        sum += arr[i];
    }
    double average = (double) sum / arr.length;
    int count = 0;
    for (int i = 0; i < arr.length; i++) {
        if (arr[i] < average) {
            count++;
        }
    }
    return count;
}