updated the test demo and commenting