Updated unit tests to work on reduced data set