Layout fixes (Edit page)