Add editing code from UX Write