Add include code in query model