add check that Theta is correct dimensions