Refactor add button in wysiwyg script