add DI

Works Fine :)