Models and database migration added