First working unit tests