Remove unused import and make minor pep8 adjustments