Updated visualization

- Changed WignerNorm to MidpointNorm as the normalization is more
general.
-Enclosed function in try statement to catch loading if matplotlib not
available.