Added partial class for setting up DI logic.