Added option to use a custom partition key, rather than the default "spray and pray"