making scatter plots in python