They've never integrated the pen with the body afaik, wife and I had the 2/3/4/5,generally we just got a case with the Spen holder.
Still though they function on the inner screen is my point, it is supported functionally.
The outer screen tho doesn't work with the pen.
Personally I went to a Pixel when I started having hinge issues on my 4, have become increasingly anti-Google so got graphene and homelabbed up.