Added relationship association to models