Update code samples in how to migrate doc