Update docstrings to reflect the new module name