Module 3 - Added associations