add model visualization parts