Added comments for the classes