some fixes to the smoke shader