Added associations to models and db