Added setContainer method to ContainerAwareTrait