Add prediction functions.