Removed imports
Imports that were used for testing have been removed