amélioration du code et ajout d'une classe de test à implémenter