Remodelage du code

J'y ai ajouté mes modifs.