updated getDimension function