Update docstrings to use correct module name