Added legend for plots