Make a scatter plot animation matplotlib