Cleaned up unused imports. (pylint)