Updated to include JUnit test
Updated to include JUnit test