Moved DB to be injected