Trying to fix problems with UI