Adding JUnit tests - oops