Add associations for all models