Bugfix/ui tweaks