Mostly cleaning up, fixing deprecation related warnings, and nicifying visualization