Adding unit tests (100% coverage!)