Afegida la funcionalitat de classificar i la d'entrenar.