update docstring per the extension package fix, refactor
2 files changed