Fixing import statements

Removing unused import statements