Can now pass in a custom Partition