Fix edit for smaller screens fix #9