Add Junit test