Optimization - fewer number of partitions
New optimization; fix <4 elems bug
Fix a nasty bug