Refactored my code
commented on the class