Add specs for association in models