Adds more unit testing logic.