Adding DI Container support