Dependency Injection changes. Added a controller