Expanded original dataset and solved problem with array reshaping