Added migration for new schema
Pep8 changes