Cleanup a little bit the code

Using a tool like `flake8` is easy to see where you don't have a consistent code style. Things I did:
  * Removed white-spaces
  * Removed useless newlines
  * Removed unused imports
  * Trim long lines to `80` characters long
  * Added new lines between classes

Nothing to harmful, just some little and simple thing to increase readability.