added associations to models