Minor demo changes