Adding graph generation