Improve edit page formatting for mobile