Cleaned up imports

Removed some unnecessart imports from Graph