Ukbb changes

[avihu111]

• descriptor to caching
• keep keys in Collate
• some other minor changes that we might want to revert or adapt.