Remove unused imports

Remove unused imports to get rid of warnings.