Updates to the algorithm/visualization