fix small UI issues