Add some docstrings and rename some things
Add some imports into the package level space