unit testing updates