Improve page edit UX