Add JUnit testing
Merge pull request #7 from lol768/unittesting

Add unit testing