Fixing color in scatter plots in matplotlib,plt.show()