Updated DB migration script