Extend schema to allow TFn
Test serialize round-trip with pr-str/load