Added a junit test