Added fcn for quick prediction