Add imports
Fix demo
Fix log
Fix log