adding back in the dimension
still working on it