Add data migration support