Remove jUnit dependency