Updated the Demo and the readme