updated
Updated U-extension implementation