Small ui improvements.