Change color in scatter plot python