added associations

was missing associations here.