Add wrapper for select2