Updated unit testing capabilities.