add in initial association tests
update association tests