Correction et améliorations du code