Final visual tweaks of visualization UI
Small improvements