Cleaned up documentation, imports.