Working prediction test function