Removed unnecessary imports

Decluttered code, removed imports from previous tests that were no longer required