Updated included demo code