Improve grammar editing UI