Minor ui fix

Two small thing that make things look nicer.