Task: get `Prediction` using `self` not from base

To allow for future flexibility, we can refer to objects using `self`
instead of explicitly referencing the base class.