Un peu de refactoring pour un code plus propre