Clean up imports
More documentation and examples