Add ContainerAware trait