Clean up visualizer

[pzhanggit]

Some functions in visualizer.py have not been used for a while and would also likely to break the code since they are outdated. We need to either delete these functions since not used or update them consistently with other part of the code.

Also, create plots only on rank 0.