added association in models