make ui neater (thanks @SteveBarnett)