Updates for new shader