Remove imports

Removed some imports from main due to not being used.