Improve modularity to split DI modules.