Remove unused imports
Remove main which does not work