Add some dependency injection.