change Test file to JUNIT