updated to new prediction