#7 - fixing flake8 errors and warnings.

Fixes #7.