added associations on models