added comment notes for class and corrected the class