Add models and associations