imporvements on ui