Updated the demonstration