Added prediction feature to regression