unit testing working
More unit tests