Remove not necessary imports

Tidy up the file the file a bit