DI 2.0.0

This needs to be updated