Add JUnit test and makes Main class more findable.