Remove NetworkX import