Adds prediction feature