added more models + migrations