Introducing unit tests