fixed some stuff in shader