A simple Padaria system kkk

Main.java

import java.text.MessageFormat;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Padaria padaria = new Padaria();
        padaria.registarProduto("Pão", 1, 1);
        padaria.registarProduto("Leite", 3, 1);

        Scanner scanner = new Scanner(System.in);

        System.out.println("Olá, temos os seguintes produtos disponíveis:");
        for (Produto produto : padaria.getProdutos()) {
            System.out.println(MessageFormat.format("• {0}: R${1}/{2}un", produto.getNome(), produto.getPreco(), produto.getUnidades()));
        }
        System.out.println("Digite o nome do produto que deseja comprar.");

        Produto produto = padaria.pegarProdutoPeloNome(scanner.next());
        if (produto == null) {
            System.out.println("Não encontramos o produto mencionado.");
            return;
        }

        System.out.println("Quantas unidades do produto deseja comprar?");
        int unidades = scanner.nextInt();
        if (unidades <= 0) {
            System.out.println("Insira uma unidade válida e acima de 0.");
            return;
        }

        System.out.println(MessageFormat.format("Você comprou {0} unidade(s) de {1} por {2}.", unidades, produto.getNome(), (produto.getPreco() * unidades)));

        scanner.close();
    }
}

Padaria.java

import java.util.HashSet;
import java.util.Set;

public class Padaria {

    private Set<Produto> produtos;

    public Padaria() {
        this.produtos = new HashSet<>();
    }

    public Set<Produto> getProdutos() {
        return produtos;
    }

    public Produto registarProduto(String nome, double preco, int unidades) {
        Produto produto = new Produto(nome, preco, unidades);
        produtos.add(produto);

        return produto;
    }

    public Produto pegarProdutoPeloNome(String nome) {
        Produto produto = null;

        for (Produto obj : produtos) {
            if (obj.getNome().equalsIgnoreCase(nome)) {
                produto = obj;
                break;
            }
        }

        return produto;
    }
}

Produto.java

public class Produto {

    private final String nome;
    private final double preco;
    private int unidades;

    public Produto(String nome, double preco, int unidades) {
        this.nome = nome;
        this.preco = preco;
        this.unidades = unidades;
    }

    public String getNome() {
        return nome;
    }

    public double getPreco() {
        return preco;
    }

    public int getUnidades() {
        return unidades;
    }

    public void setUnidades(int unidades) {
        this.unidades = unidades;
    }
}