Add associations to factory