added unit tests. fixed include on demo scripts