Add association to models