Added DI code for store