Fix for issues1 - added JUnit tests

[jsangeetha]

Hi,
For the [Issues1 | https://github.com//issues/1], added Junit Tests along with the fix.

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}