Add ability to specify policy in graph