more unit testing work