Update code to v1beta2

+1 Sid.