Add graph generation