add junit support