DI 2.0.0

Move these benchmark changes to a PR on top of DI 1