JUnit tests (Forgot to add "test" folder on last commit)