Merged changes from "Assoc update cleanup and added testcases" PR

Hi @junmendoza, please have a look. Thanks!