Update JUnit testing