Added test data for part 2