Update do_migration() method to run DB migration