Removing unused import and fixing pylint violation
Fixing pylint violation