removed unused import
added NOQA to silence flake8 about the import statements