Add missed comments