Trying to implement unit testing...