major changes to prediction model