UI fixes

Even if it doesn’t work it is sexy