Corrected errors in the demo