Remove old code from graph model