Made demo code clearer