Ajout d'un Model + petite restructuration du code