Add models and db creation