Got JUnit to work