Added an option for the amount of partitions

This may be helpful. Was for me.