Add constructor injection