Legend added to plots

A simple legend was added to the plots.