Add container aware trait