added small UI fixes