Changing it to scatter plot