Add simple clustering to regression