Added association to Apartment model