Added DI module to the mix