Changing demo a bit to make more sense