Updated SimpleJob to also handle partitioned steps