Added simple JUnit tests - fix #5