Make legend for scatter plot matplotlib