improved the editing screen