reverted to old graph for testing