changed the predict the way it passes in references