Added some basic UI to the demo