Add comments for classes