Update get_centroid.py