Minor changes in vis shader