Step 2: Rewrite update_assignment When we implemented the update_assignment function, it assigned each data point to the centroid it is closest to. This works great for our algorithm, but now we are not only interested in which centroid the data point is assigned to. but also the true label of that data point in order to perform the accuracy analysis. Therefore, we need to update the function to incorporate more parameters and return different values. Implement the following function in analysis.py: def update_assignment(data, labels, centroids): """Assign all data points to the closest centroids and keep track of their labels. The i-th point in "data" corresponds to the i-th label in "labels"