I really don't like type matching here. How about we actually encode the behavior on the type of classification object being returned?