added few unit tests