Changing MultiModel class to return in-memory list of predictions