Создание кнопки в Tinymce Wordpress

Привет. Есть необходимость создать свою кнопку в Tinymce, кнопка должна обрамлять тегами <div> выделенный участок текста. С этим проблем нет, кнопка создана и добавлена в панель редактора. Вопрос в другом, как при повторном выделении этого блока (очистить форматирование) что бы не лазить в исходный код. В стандартных кнопках редактора при выделении текста подсвечивается тег которым он обрамлён , а в созданных пользователем нет. Как это исправить! Спасибо!

Код js скрипта. Автор: SEO Маяк

(function() {

 tinymce.create('tinymce.plugins.mayak', {

 init : function(ed, url) {

 ed.addButton('yellow', {

 title : 'Желтый фон',

 image : url+'/images/Название изображения.png',

 onclick : function() {

 ed.selection.setContent('[yellow]' + ed.selection.getContent() + '[/yellow]');

 }

 });

 },

 createControl : function(n, cm) {

 return null;

 },

 });

 tinymce.PluginManager.add('mayak', tinymce.plugins.mayak);

})();

Код в function.php

function mayak_button()

 {

 if ( current_user_can('edit_posts') && current_user_can('edit_pages') )

 {

 add_filter('mce_external_plugins', 'mayak_plugin');

 add_filter('mce_buttons_2', 'mayak_register_button');

 }

}

add_action('init', 'mayak_button');

function mayak_plugin($plugin_array){

 $plugin_array['mayak'] = get_bloginfo('template_url').'/js/newbuttons.js';

 return $plugin_array;

}

function mayak_register_button($buttons){

 array_push($buttons, "yellow");

 return $buttons;}