Migrations and small fix
__init__