Legend colors fix

Fixed the colors of the legend so they are displayed correctly.