Added first JUnit tests