update UI to match with class structure digger