Add support for select2