Add module docstring
Cleanup whitespace