NOTEBOOK: Add necessary import for graphX