Added plot to surf_predict