Didn't actually add JUnit tests. (OOPS!)