Add option to add class to the container