Updated to junit4 test suite