WIP - Implement migration steps