Dúvida Java trabalhando com listas

Boa tarde!! curso: Java trabalhando com listas e coleções de dados 08Desafio: hora da prática Questão: Crie uma classe Produto com propriedades como nome e preço. Em seguida, crie uma lista de produtos e utilize um loop para calcular e imprimir o preço médio dos produtos.

Uma dúvida o meu código ficou diferente da aula poderia comentar?

Segue a resposta da aula: import java.util.ArrayList;

class Produto { private String nome; private double preco;

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

public String getNome() {
    return nome;
}

public double getPreco() {
    return preco;
}

}

public class Main { public static void main(String[] args) { Produto produto1 = new Produto("Ar condicionado", 2000.0); Produto produto2 = new Produto("Aquecedor", 800.0); Produto produto3 = new Produto("Ventilador", 150.0);

    ArrayList<Produto> listaProdutos = new ArrayList<>();
    listaProdutos.add(produto1);
    listaProdutos.add(produto2);
    listaProdutos.add(produto3);

    double somaPrecos = 0;
    for (Produto produto : listaProdutos) {
        somaPrecos += produto.getPreco();
    }

    double precoMedio = somaPrecos / listaProdutos.size();
    System.out.println("Preço médio dos produtos: " + precoMedio);
}

}

Aqui segue a minha solução:

package br.com.screenmatch.desafio8Lista.produto; public class Produto { private String nome; private double preco;

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

public String getNome() {
    return nome;
}

public void setNome(String nome) {
    this.nome = nome;
}

public double getPreco() {
    return preco;
}

public void setPreco(double preco) {
    this.preco = preco;
}

@Override
public String toString() {
    return "Produto : "+nome+" Preço: "+preco+"\n";
}

}

package br.com.screenmatch.desafio8Lista.produto; import java.util.ArrayList; public class TestaProduto { public static void main(String[] args) { ArrayList produtosLista = new ArrayList<>(); produtosLista.add(new Produto("CEL",2000)); produtosLista.add(new Produto("Table",980));

    System.out.println(produtosLista.toString());

    double somaPrecos = 0;
    for (Produto produto : produtosLista) {
        somaPrecos += produto.getPreco();
    }

    double precoMedio = somaPrecos / produtosLista.size();
    System.out.println("Preço médio dos produtos: " + precoMedio);
}

} Desde já agradeço.