Modify bug, completed the demo of the program