improvements in UI