better ui on modal