work on flake8 compliance and test coverage