added prediction test