fixed a few UI issues