Remove not necessary imports

Tidy up the file a bit