Add a legend and colored circles for focus