Added legend to plot (weighted average still WIP)