Get average model visualisation working.