Code now does simple clustering, still very hardcoded though