Cleaned up code

removed some unused imports