Fix class doc comments