Implementing better unit testing