Implementing unit testing