New gen and U-Net architecture