Improved unit tests system