Updated the function for partition sampling.
More (minor) changes to the partition sampling subroutine.