Added models to class and migrations.