imrpoved UI