Adding JUnit Testing