Changed the partition function so that the lab will pass.