[IMP] better ui