I've updated the visualization to correct all the mistakes it was making