Updated WFS point source driving function