Add belongs to models
Add has many models