`transform()` should return same type as callback fn

[mattvague]

Right now transform always return MathNode. Would be better if it returned the type returned by its callback