small ux improvements