Add prediction related functions

So far only get_prediction_variables which was not committed before