container element now accepts class param