Add graph generation for simple cases