Add logging to demo