1. added code to predict next set of points