Minor docstring fixes and removal of unused imports

This PR makes some minor docstring improvements and removes unused imports