putting Graph class back in for component checking