Predict the output | StemPy Lab