Adding code for Grace (DI Container)