refactor: Put the demo code back

Lol...