Edited the demo to show the new features