Fixed Pyflake unused import warnings