Add prediction functions