adding data points code back in