Move tests to JUnit
Closes #13.