Don't load visualization until needed

Visualization functions should load only when needed (and conditional
imports are already in the right locations), so remove the imports from
the top of the main module.