Be able to override default template.

Currently we are forced to use a `div` as container.
This allows to specify whatever we want.

Thanks!