More small UI updates