Add ability to generate graphs