Added simple test using JUnit.