Implementing unit tests