Improve visual model explanation