improve UI layer