add prediction functions