fixed the unit tests
added demo and such