#3 Connect JUnit and add simple test