Renamed predict to prob