Made some changes to vis 2