Base on jupyter/demo