ui improvements & sizing etc