made changes to predict