Improved UD word collection