Update the migration script

Bring the migration script up to date.

Change-Id: I8f559be7bdc53a05f4700a623e83975e606edcf4
Signed-off-by: Ryan Barry <rbarry@redhat.com>