Small changes to visualziation.