now building off of StatsBase.predict()