Add test for U and WU