Adds prediction support