Added very basic JUnit tests