updated test to Junit4