big update to the demo