Correction des bugs restants + améliorations