Add some tracing to demo