Improve editing screen on mobile