modify junit to test