remove comment here, add docstring in class definition