Add demo.py. Correct some function name