1. Add the legend part