Visualization coloring should now be more accurate! Uses dist now