clustering working
fixed main