Added comments for class