Adding all the code for the demo stuff