A bunch of viz fixes to make things prettier