Fix class comments