updating to better match the prototype
removing kmeans shape detection