updated header comment
changed demo code a bit