Improved unit testing framework