Slight refactor
Slight refactor

kotcrab/vis-editor,billy1380/VisEditor,StQuote/VisEditor,kotcrab/vis-editor,piotr-j/VisEditor,code-disaster/VisEditor,kotcrab/VisEditor,intrigus/VisEditor,piotr-j/VisEditor