Как лучше доработать код?

import java.util.HashMap;
import java.util.ArrayList;
 
public class ExpensesManager {
    HashMap <String, ArrayList<Double>> expensesByCategories; // Замените на таблицу с именем expensesByCategories
    ArrayList<Double> expenses; 
    
    ExpensesManager() {
        expensesByCategories = new HashMap<>(); // Создайте таблицу
        expenses = new ArrayList<>();
    }
        
        // Добавьте в метод ещё один параметр — category
    double saveExpense(double moneyBeforeSalary, double expense, String category) {
        moneyBeforeSalary = moneyBeforeSalary - expense;
        expensesByCategories.put(category, expenses); 
        System.out.println("Значение сохранено! Ваш текущий баланс в рублях: " + moneyBeforeSalary);
        
        // Замените на работу с таблицей
         if (expensesByCategories.containsKey(category)){ // Проверьте наличие категории
            expensesByCategories.get(category); // Получите список трат в этой категории
            expenses.add(expense);
            System.out.println(expensesByCategories);
             // Добавьте трату
        } else {
            ArrayList<Double> newExpenses = new ArrayList<>();// Создайте новый список трат и добавьте в него сумму
             newExpenses.add(expense);
             expensesByCategories.put(category, newExpenses); // ?
             // Сохраните категорию и новый список трат в хеш-таблицу
             System.out.println(expensesByCategories);
        }
        if (moneyBeforeSalary < 1000) {
            System.out.println("На вашем счету осталось совсем немного. Стоит начать экономить!");
        }
        return moneyBeforeSalary;
    }
    
    
    void printAllExpensesByCategories() { 
                // Замените логику для работы с таблицами
        
        for(String categoryes : expensesByCategories.keySet()){         
            System.out.println(categoryes);
        for (Double list : expensesByCategories.get(categoryes)){
        System.out.println(list);
                 
             }                  
        } 
    }
    
        // Метод должен принимать название категории и называться findMaxExpenseInCategory
    double findMaxExpenseInCategory(String category) {
        double maxExpense = 0;
        /* Замените логику для работы с таблицами
        Если категория есть, то ищем максмальную трату.
        Иначе печатаем "Такой категории пока нет." */
        if(expensesByCategories.containsKey(category)){
            for (Double exp : expenses) {
            if (exp > maxExpense) {
                maxExpense = exp;
            }
            }            
        } else{
             System.out.println("Такой категории пока нет.");            
        }
        return maxExpense;
    }
 
    
 
    void removeAllExpenses() {
        expensesByCategories.clear(); // Таблица называется иначе
        System.out.println("Траты удалены.");
    } 
}

import java.util.Scanner;
 
public class Praktikum {
    public static void main(String[] args) {
 
        Scanner scanner = new Scanner(System.in);
        System.out.println("Сколько денег у вас осталось до зарплаты?");
        double moneyBeforeSalary = scanner.nextDouble();
        System.out.println("Сколько дней до зарплаты?");
        int daysBeforeSalary = scanner.nextInt();
 
        Converter converter = new Converter(78.5, 88.7, 0.75);
        DinnerAdvisor dinnerAdvisor = new DinnerAdvisor();
        ExpensesManager expensesManager = new ExpensesManager();
 
        while (true) {
            printMenu();
            int command = scanner.nextInt();
 
            if (command == 1) {
                System.out.println("Ваши сбережения: " + moneyBeforeSalary + " RUB");
                System.out.println("В какую валюту хотите конвертировать? Доступные варианты: 1 - USD, 2 - EUR, 3 - JPY.");
                int currency = scanner.nextInt();
                converter.convert(moneyBeforeSalary, currency);
            } else if (command == 2) {
                dinnerAdvisor.getAdvice(moneyBeforeSalary, daysBeforeSalary);
            } else if (command == 3) {
                System.out.println("Введите размер траты:");
                double expense = scanner.nextDouble();
                System.out.println("К какой категории относится трата?");
                String category = scanner.next();
                moneyBeforeSalary = expensesManager.saveExpense(moneyBeforeSalary, expense, category);
            } else if (command == 4) {
                expensesManager.printAllExpensesByCategories();
            } else if (command == 5) {
                System.out.println("В какой категории искать?");
                String category = scanner.next();
                System.out.println("Самая большая трата в категории " + category + " составила "
                        + expensesManager.findMaxExpenseInCategory(category) + " руб.");
            } else if (command == 6) {
                expensesManager.removeAllExpenses();
            } else if (command == 0) {
                System.out.println("Выход");
                break;
            } else {
                System.out.println("Извините, такой команды пока нет.");
            }
        }
    }
 
    public static void printMenu() {
        System.out.println("Что вы хотите сделать? ");
        System.out.println("1 - Конвертировать валюту");
        System.out.println("2 - Получить совет");
        System.out.println("3 - Ввести трату");
        System.out.println("4 - Показать траты по категориям");
        System.out.println("5 - Показать самую большую трату в выбранной категории");
        System.out.println("6 - Очистить таблицу трат");
        System.out.println("0 - Выход");
    }
}