Correction in classes [style and design]