Cleaned up imports
Formatted code
Updated documentation.