UI cleanup per Samir