fix class comments