add junit test