Clean up & ajout de tests