added a few unit tests