Please add a comment here explaining why you're explicitly setting the number of partitions. Will be really useful for the next person looking at the code.