Added demo code (hack)