added small UI features