Added DI for the db context