Amélioration du code

Moins de bordel ~