Added setting the custom container