PEP8 sumpy.visualization.
Fix visualization
More vis fixes