add junit testing