refactor: Remove demo code