Initial attempt at integrating graph code