Better layout for edit page