[FEATURE] Add legend to graph