Implement unit tests