Added data set dimensions to expected output