Add keep_undim parameter to extract_udtype