Update function not returning the right number of dimensions yet.