Cleanup code.
Reorganize imports
Add some description and module information.