- update for new graph structure