Added scatter plot code