Fixed warnings by removing unused imports