Final ui tweeks