another tweak to the yhat demo