UI fixes

Ass hole fix