I think the pumping smoke looks great, and probably is a more balanced visualization. Now it is possible to actually see :-D.
I made a small change rather late yesterday and it seems like you missed it; line 346 in control.lua:
elseif entity.type == "boiler" then
became:
elseif entity.type == "boiler" or entity.type == "burner-generator" then
burner-generator is a vanilla type, but is not used in any vanilla entities, but for AAI burner turbine it makes it so the smoke is scaled like for boilers.
If you make an update later you might want to include it.