Update JUnit Testing