PEP8 fixes on migration script