improve styling and ux