Samll UI improvment