Make Diagrams.Transform.under more polymorphic
Merge pull request #168 from FlorentBecker/master

Make Diagrams.Transform.under more polymorphic