Small cosmetic updates