UI fixes

IDIOT