add associations logic