Add associations to models