Code cleanup, fixed toolbar visibility

Toolbar button should only be visible in editor or KSC.