#105 Refactor the WYSIWYG