starting to implement unit testing