Add explicit return type