Add better association model info