Implemented DI
Implemented basic test with DI