Improvements for editor
Visual improvements to the editor.
Quick bugfix