Make demo actually work