Docs minor fixes and import removes

This PR makes some minor docstring improvements and removes unused imports