add junit test and comment