Added JUnit testing
Added JUnit testing