Small UI fine tuning, thanks to Ajaska and savpek!