Fixed dimensionality problems, now runs correctly