Added a legend
Fixed legend