Updated predict to use the new Theta structure.