Fixed pyflake import warnings