Add models when creating DB