fixed wrong docstring of Unit

Fixes incorrect docstring and PEP-8 fixes the module.