Custom partitioning example